import requests
from scrapy.selector import Selector
import MySQLdb

conn = MySQLdb.connect(host="127.0.0.1", user="root", passwd="123456", db="search_spider", charset="utf8")
cursor = conn.cursor()


def crawl_ip():
    headers = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/78.0.3904.97 Safari/537.36"}
    for i in range(1, 4039):
        re = requests.get("http://www.xicidaili.com/nn/{0}".format(i), headers=headers)
        selector = Selector(text=re.text)
        all_trs = selector.css("#ip_list tr")

        ip_list = []
        for tr in all_trs[1:]:
            speed_str = tr.css(".bar::attr(title)").extract()[0]
            if speed_str:
                speed = float(speed_str.split("秒")[0])
                if speed > 2:  # 速度大于2秒的不要了
                    continue
            all_texts = tr.css("td::text").extract()

            ip = all_texts[0]
            port = all_texts[1]
            proxy_type = all_texts[5]  # 判断代理类型
            if proxy_type != 'HTTP' and proxy_type != 'HTTPS':
                continue
            ip_list.append((ip, port, proxy_type, speed))

        for ip_info in ip_list:
            cursor.execute(
                "insert proxy_ip(ip, port, speed, proxy_type) VALUES('{0}', '{1}', {2}, '{3}')".format(
                    ip_info[0], ip_info[1], ip_info[3], ip_info[2]
                )
            )
            conn.commit()

class GetIP(object):
    def delete_ip(self, ip):
        # 从数据库中删除无效的ip
        delete_sql = """
                    delete from proxy_ip where ip='{0}'
                """.format(ip)
        cursor.execute(delete_sql)
        conn.commit()
        return True

    def judge_ip(self, ip, port):
        # 判断ip是否可用
        headers = {"User-Agent": "Mozilla/5.0"}
        http_url = "http://www.baidu.com"
        proxy_url = "http://{0}:{1}".format(ip, port)
        try:
            proxy_dict = {
                "http": proxy_url,
            }
            response = requests.get(http_url, proxies=proxy_dict,timeout=1,headers=headers)
        except Exception as e:
            print("invalid ip and port")
            self.delete_ip(ip)
            return False
        else:
            code = response.status_code
            if code >= 200 and code < 300:
                print("effective ip:"+proxy_url)
                return True
            else:
                print("invalid ip and port")
                self.delete_ip(ip)
                return False

    def get_Random_Ip(self):
        # 从数据库中随机获取一个可用的ip
        random_sql = """
                      SELECT ip, port FROM proxy_ip 
                    ORDER BY RAND() 
                    LIMIT 1
                    """
        result = cursor.execute(random_sql)
        for ip_info in cursor.fetchall():
            ip = ip_info[0]
            port = ip_info[1]

            judge_result = self.judge_ip(ip, port)
            if judge_result:
                return "http://{0}:{1}".format(ip, port)
            else:
                return self.get_Random_Ip()

if __name__ == "__main__":
    get_ip = GetIP()
    get_ip.get_Random_Ip()

# get_ip = GetIP()
# print(get_ip.get_Random_Ip())